# _*_ coding: utf-8 _*_
"""
@ 时间    ：2024/10/23 16:05
@ 作者    ：旺财
@ 文件    ：01 随机森林的分类与回归模型.py
@ 说明    ：随机森林的分类与回归模型
"""

from sklearn.ensemble import RandomForestClassifier, RandomForestRegressor

"""随机森林是常见的集成学习模型Bagging算法中非常典型的机器学习模型"""
# 1.分类模型
X = [[1, 2], [3, 4], [5, 6], [7, 8], [9, 10]]
y = [0, 0, 0, 1, 1]

model = RandomForestClassifier(n_estimators=10, random_state=123)
model.fit(X, y)

print(model.predict([[5, 5]]))

# 2.回归模型
X = [[1, 2], [3, 4], [5, 6], [7, 8], [9, 10]]
y = [1, 2, 3, 4, 5]

model = RandomForestRegressor(n_estimators=10, random_state=123)
model.fit(X, y)

print(model.predict([[5, 5]]))